from torch import nn
import torch
import torch.nn.functional as F

class TuDu(nn.Module):
    def __init__(self):
        super(TuDu, self).__init__()
        

    def forward(self, x):
        output = x + 1
        return output

tudu = TuDu()

x = torch.Tensor([1, 2])

output = tudu(x)
print(output)